package com.wxweven.algorithm.array;

import java.util.Stack;

/**
 * @author wxweven
 */
public class LC503下一个更大的元素 {

    /*
     * 给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。
     * 数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。
     * 如果不存在，则输出 -1。
     *
     * 示例 1:
     *
     * 输入: [1,2,1]
     * 输出: [2,-1,2]
     * 解释: 第一个 1 的下一个更大的数是 2；
     * 数字 2 找不到下一个更大的数；
     * 第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
     * 注意: 输入数组的长度不会超过 10000。
     * 讲解地址：https://leetcode-cn.com/problems/next-greater-element-i/solution/dan-diao-zhan-jie-jue-next-greater-number-yi-lei-w/
     * 视频地址：https://www.youtube.com/watch?v=5IAq5jd8O7Y
     *
     * 思路：单调栈
     */

    public int[] nextGreaterElements(int[] nums) {
        // 边界条件
        if (nums == null || nums.length == 0) {
            return new int[0];
        }

        int len = nums.length;

        int[] res = new int[len];
        Stack<Integer> stack = new Stack<>();

        // 必须是从后往前遍历
        for (int i = 2 * len - 1; i >= 0; i--) {
            /**
             * 如果栈顶元素比当前小，那么就一直出栈
             */
            while (!stack.isEmpty() && stack.peek() <= nums[i % len]) {
                stack.pop();
            }

            // 如果栈为空了，说明没有比当前元素大的，否则栈顶就是比当前元素大的第一个元素
            res[i % len] = stack.isEmpty() ? -1 : stack.peek();

            stack.push(nums[i % len]);
        }

        return res;
    }
}
